package com.yhsimon.flyweight;

import java.util.HashMap;

/**
 * @author： YHSimon
 * @date： 2021-10-04 18:36
 */
//工厂类，根据需求返回一个网站
public class WebsiteFactory {
    //集合，充当池的作用
    private HashMap<String,ConcreteWebsite> pool=new HashMap<>();

    //根据发布形式返回一个网站，如果没有则创建一个网站并放入池中，并返回
    public Website getWebsiteCategory(String type){
        if(!pool.containsKey(type)){
            pool.put(type, new ConcreteWebsite(type));
        }
        return pool.get(type);
    }

    //获取网站分类的总数(池中有多少个网站类型)
    public int getWebsiteCount(){
        return pool.size();
    }
}
